#!/usr/bin/env python3
# -*- coding: UTF-8 -*-

class Config:
    LOAD_MODEL_ID = 0 #1753896523
    
    # Core RL parameters
    GAMMA = 0.99
    TDLAMBDA = 0.95
    START_LR = 0.0003
    BETA_START = 0.0005
    BETA_DECAY = True
    CLIP_PARAM = 0.2
    VF_COEF = 0.5  # Updated as recommended
    
    # Actions
    ACTION_LEN = 1
    ACTION_NUM = 16
    
    # In conf.py - update these values:
    SPATIAL_FEATURES = {
        'height': 51,
        'width': 51, 
        'channels': 2  # Reduced from 3 to 2
    }

    SPATIAL_FEATURE_SIZE = SPATIAL_FEATURES['height'] * SPATIAL_FEATURES['width'] * SPATIAL_FEATURES['channels']  # = 5202
    NON_SPATIAL_FEATURE_SIZE = 27  # Much reduced from 53
    FEATURE_LEN = SPATIAL_FEATURE_SIZE + NON_SPATIAL_FEATURE_SIZE  # = 5229
    
    
    
    # Calculate total feature dimensions
    
    
    TOTAL_FEATURE_SIZE = SPATIAL_FEATURE_SIZE + NON_SPATIAL_FEATURE_SIZE
    
    # Update legacy config for compatibility
    FEATURES = [TOTAL_FEATURE_SIZE]
    FEATURE_SPLIT_SHAPE = FEATURES
    FEATURE_LEN = TOTAL_FEATURE_SIZE
    VALUE_NUM = 1
    
    DATA_SPLIT_SHAPE = [
        FEATURE_LEN,  # features
        VALUE_NUM,    # reward
        VALUE_NUM,    # old_value
        VALUE_NUM,    # tdret
        VALUE_NUM,    # adv
        ACTION_LEN,   # old_action
        ACTION_LEN,   # old_prob
        ACTION_NUM,   # legal_action
    ]
    data_len = sum(DATA_SPLIT_SHAPE)
    SAMPLE_DIM = data_len